Removed all the id and css using class instead. Refactored some logic.