Nicer looking plots for thesis using matplotlib