Budget Analysis Python Code