Adding assertEquals function