reformat
use array comprehensions
Merge pull request #2 from sailxjx/master

use array comprehensions