Update

Small fixes