Adds docs, fixed models