Update + fix bugs