Fixed stupid demo bugz... 2 if-statements :(