Change import style for __init__.py

Ori pointed out in I5c6b739 that he would prefer a single import listing
all imported classes rather than a separate import per class.

Change-Id: I467511acc43bb19377bc05a133a158aeba1fbaf3