fixing class assignment, done for the night