Simplifying Code and fixing ball population