Simplifying UI, updating styles to make it purtier