Initial Value Problem in Python