Khairuddin happy with much-improved KTM