Second homework done. Working with minimal testing