Merge pull request #9 from stuycs-ml7-projects/sofia

forgot to commit yesterday
Merge pull request #10 from stuycs-ml7-projects/sofia

addclass