code update bug fixes