Lab 6 Milestone 1 better version