Added some styling to make it look less janky