Small UI improvement for the 2. extra credit