completed assignment, just need to add comments