submission of second hw assignment

You can just do `if function_desired` here - no need to introduce another variable.