Fixed naming to be in sync with homework task