minor refactor and added to readme