replaced function comments with doc strings
did some code cleanup post in-class review