add small changes to functionality