Added better variant of homework