This seems reasonable. 

Though not necessary, would it be easier to grab that info when the `current` class is set rather than after the fact?