updates / bug fixes