Don't forget to use the right class for before and after methods, too