Fixing comment position

Code aesthetic, just fixing comment position to where they should be.