Lab 4 Milestone 2 fixed