updates, couple bugs left