Improve plot legend
Enable prerec fitness function support