Better docs, fixes and more functionality