Code Cleanup :cat: