Update

Made it more stable