Update and fix documentation