done adding the new features, time to clean up the code and change the documentation to fit the new functionality