Updates TODO
Updates tests