adding slight functionality