Sprint 3 : Tests + Bug fix