Return statement updated
Working digital walle