Fix simple image flipper example

Fix states as well