Updated failing test to assignment 2