part 2 working on display logic