trying to fix python code block