Great functionality! New features wanted.

It worked perfectly with my current site but I found one small issue

When typing in the /wp-login.php or /wp-admin on a network site, it will redirect to the properly set main site, but when logging in, it does not redirect back to the original page and instead presents you with the main sites’s /wp-admin page