Ball Cleanup

Added clean up function to ball states.