Example code for using Flux with EECS 470