get_extension now returns extension of any file, not just those with three letters

Also, fixed spacing for comments on line 21-22 and 27-28.