Changed the functions for gridworlds so that takeAction returns the reward. Seems to be a bit more natural