Add compute rotaiton function