Sunderland boss Martin O'Neill is looking for Adam Johnson to kick on after finally scoring his first goal for the club.