made colored balls a separate class