Re-write StaggeredGridView layout code.
Remove unused import.