Updates & bug fixes