Cleanup function, fix #11