Minor polishing

Just a little bit more polishing