Changed the .h file -- step 6