HW3EX1 working

:
Add test files
Fix type signatures
Small refactor