Added comments.

    Still need to add unit tests