small change for in wiriting code