Removed unecessary class