Added CSS styling and removed button element