Add mask size for more modularity

[albertodallolio]

Hey, I added the possibility of changing the mask size in update.py module. This could give more modularity for further testing and development.

Thanks