clean up and make code more readable
add info to readme