Fixed code formatting
Updated documentation