Added end brace to cleanup function
added end braces
Merge pull request #2 from thosey/master

Added end brace to cleanup function