Fixes the segfault

Now just dealing with some wrong answers.