Add rotation logic