Add some todos
Add colour modes for return index and number of returns