Added documentation.
Added output reshaping. Still need to add tests to check for it.
Added small detail to doc.