refactor: Cleaned up code