Use diff class