Ball fixes

Fixed ball functionality