Fix bug in state drawing
Merge pull request #8 from pthedinger/master

Fix bug in state drawing