Add voltage enum
Remove previous hack