Adding code for wallet implementaiton