Fixing code so balls drop!