Added wrap class for smoother transitions