Small changes to info-display solution