My version of assignment 1!