Cleaning up the code lines