Improved USB callback functionality (#13)