Minor visual changes to the code