Adjusting code
Updated documentation