The National Museum of Scotland has been served an improvement notice by health inspectors investigating the outbreak of Legionnaires' disease in the city.