added comments, completed assignment