Removed unneccesary class