Added goToJail function