Bug fix, added files, added info to readme