Add over current detection