Add tests for For