Clean up. update readme