Cleaned up module import. I recognize this isn't the most optimal solution out there; it's redundant checking every integer in the range, but for the small maximum (20) as prompted, I'll pick elegance over efficiency for the time being. I might revisit this problem later.