Fixed the doctest and removed unecessary comments