Add cleanup

Signed-off-by: Paul de Wouters <paul@hmn.md>