Fixed assertRaises error. Program now works.