Commiting HW7 with list comprehention stuff.... wooo...!!