Added code, modified readme