Edited the context function so that it finds words before and after

And generalized it so that one can find ‘x’ words around it