Update and add documentation