A simple Flux implementation