Added comments. More pythonic-ifying