Completed if else assignment