working on bug still