cleaned up code formatting