Remove TODO stuff