N mod M easy challenge solved in python