Working on simple functionality of classes