Cleaned up code & add a new function for god mode