add cleanup

Signed-off-by: Micky Ching <mickyching@gmail.com>