More refactoring ang cleanup