Cleans code with migration update