Redid program to be idiomatic(AKA amazing)