Improve code

in 37 seconds, using 0 compute credits, and was queued for 3 seconds