Refactor code, implement actual modules