python code for heart pattern