Change main logic