Copy matrix and cleaner code... TODO: Use list comprehensions