Fixed code formatting somewhat