Just a simple model in Python