Adding enable disable functions