Update - Bug fixes