Feature/add ball rect