Fix dimensions and add TODO

in 24 minutes and 29 seconds, using 0 compute credits, and was queued for 3 seconds