1st submission python/matrix