Documentation and add TODO