More comments

Added some more comments simplified final return