Minor changes on displayed code