Noobie problem for Python, personally think the code is clean :)