Added TODO after discussion with professor