Updated for depth sensor