Added more robot functionality and updated tests to match it