Initial stab using Flux.