fix: small fixes and updates