Documentation

Updated to be in line with the code