Fixed code formatting issues