Added functionality to rotate around specified axis