forgot to add main logic