Simple tests for the homework
Add first and second func from the solution