Splitting it into two code blocks