Added ball functionality