Fix formatting and add TODO