Visual improvements and styling