Klein returns to K-State as QB coach