cleaned up code, added github link