You have to use the global variable `$custom_theme`.  
The implementation of your `hook_init()` is good.