Refactor for better alignment, annotation and a bit smaller.

[Montana]

A little refactor, to make the file smaller, a bit better for making annotations in the future (if need be.)

[palantirtech]

Thanks for your interest in palantir/python-language-server, @Montana! Before we can accept your pull request, you need to sign our contributor license agreement - just visit https://cla.palantir.com/ and follow the instructions. Once you sign, I'll automatically update this pull request.