Add more test to project
Refactor tests
Test empty milestone