Added basic functionallity to the class