How can I optimize my code and make it less chaotic?