Return of serve is essentially a ground stroke, so improving your ground strokes will naturally make your return of serve better. The only difference is the serves may come at you at a greater speed than normal strokes. As a result, things like split steps, moving forward, cutting off angle, shortened backswing…become much more important in your preparation.
Knowing exactly what to do with a certain kind of serve will help. For example, I treat a high kick serve to the BH as a looping ball ground stroke to my BH, so I hit it on the rise, slice a high BH, or lob.