Separate functions, add theme to plots