adding functionality fixes to first test