Add readme and update .py