More cleanup

Just updating comments