Small ball fixes
Merge pull request #2 from Neob91/small_fixes

Small ball fixes