Made small changes to code format