1st submission python/difference-of-squares