Fixing bad code formatting