added minor functionality