Moved code block to proper place!