clean code & update readme