Working on exercise set 3, multigrid with Claudio Llinares