Fixed code formatting problems