Added the Display class