Classes are working properly