Code for Milestone 1