Small corrections with adding documentations