Update readme, add TODO