Moved some code and updated the styling for the a element