Improved organization: Separated code into sections