added todos in tests