Overall simplification of the code

[sylhare]

I wanted to see how I can make the algorithm still entirely readble while taking less space

Using:

• maps, filter, lambdas when applicable
• restructuring the code zhen necessary
• one liner ifs.