Demo test functionality added