These bugs are moving in different ways and it's a lot of lines of code! Can you modify the code to make the program more simple and efficient?