updated flex box, done for now