- Create utilities for moving/placing the object at any position
- Checking if the gripper position is on the object
- Change plotting (figure out how to represent progress in a graphical way)
- Try soft target update
- Set up replay buffer to hold augmented data
- Rotations can be done later, if we still need performance improvements
- Random image backgrounds