Fixing the display issue! woohoo!