Improved docs, fixed identation