Add in_words()
Remove unused imports from maths and add flatten()