Enhancing Neural Code Representation with Additional Context