Removed the unnecessary class