Python check output