Corrected and added documentation