Implemented class assigning - the fancy way :D