a llittle bit modfication