prettified code from the first weeks assignment