Included display class in document generator. Not my proudest moment.