functionality works, just need to style