Add direction class to the existing classes