Minor Fix And Added Comments