Small Improments

Fixed a README bug, and funtionalized some of the code. Can squash if needed.