Finally, actually got class changing to work