Added first test. Work on tests