Adding additional test cases (with incorrect outputs from class)