adding test functionality