Hopefully fixed the display bug
Merge pull request #2 from hedgehog1029/patch-2

Hopefully fixed the display bug