made custom matplotlib theme while model is running