Minor changes in code format