Update README, add TODO