update after last review

Signed-off-by: Erik Bernoth <erik.bernoth@gmail.com>