Modify to use depth camera