First working test for Milestone 2