trying to add functionality