Adds functionality for class