updated code

Signed-off-by: David Fauth <dsfauth@gmail.com>