Clean up baseline model