Added code, updated readme