Refactor: Remove unused files and some tests