Fix to balls demo 2