Adding more JavaDocs. Also got rid of the InitialState object. No need to have it anymore.