Fixed styling and added functionality.