Now separate logic for every element