Refractor device image registration into function calls

[sherry-yuan]

Group the calls here into function instead of just wrapping them in a clause.

Each section that begins with // ... and do sth should be in its own function

Assignees: @sherry-yuan