Add keys() to Flow class