fixing some code formatting