more code, updated readme