small changes to prototype