HSSP week 5 code for class