Updated problem defition