cleanup : imports
cleanup: doc strings and minor refactoring
cleanup: docstring, single/double quotes, remove unused code