Added logic to the green block