Update

Just a few tweeks