Intial code for Milestone 3