Fixed code formatting
Added some docs