Add better styling