updated code

Signed-off-by: Andrew Miller <amiller@dappervision.com>