added reg button
minor fix
Final version