Test classes

Something like this?