Add check for correct no. of args to each function
Make colour channel optional