Removed the unnecessary Callback class