rewrote class to handle two balls