chore: minor improvements