Changes at the end of class (not working)