Style the code a bit