Added switch for diff functionality