Final fixes!!