Updated Code

Fixed a display issue