tidy up

Signed-off-by: jasonh1 <jasonho.ib@gmail.com>