Feature lattice2

I have addressed the issues. Can you all take another look.Thanks.
It is highly uncommon to have such comments in python. The end of the code block is designated by indention and such comments do not provide any additional info. I would suggest to remove them to avoid them to reduce clutter.