Modified to work with 5V input