add first draft of ball logic