Introduce python typing specifications

• Introduce typing in functions arguments
• Speed up some for loops
• Reorganize the code to shorten some functions etc