Removed problem class
Updated ball to use doubles