Added code, improved readme