Added functionality to gem