add TODO: Make it a type! :)