Some pythonic suggestions