P04: using shorter function def
P05: Done