Refactor code
Changed css style