Added code for assessment 1