added return class
Merge branch 'master' of https://github.com/kenzo00/CPSC304_Project
Merge branch 'master' of https://github.com/kenzo00/CPSC304_Project
Return class