Slightly simplified code

Cannot believe I was using the wrong email to submit all of these
commits. It wasn't showing my progress on the graphs. Should be fine
now.