Change ball addition in demo