update + bug fixes