Add and fix documentation