updated comments and driver file