Starting work on the automation section

There is probably a cleaner way of doing this, but I'm new to Python