Removed unused code and started updating docs