minor changes from class

couple tings