Added assertEquals to test file