Changed code formatting
Added some docs