Move check for root user into python script