Working on Test Driven Development