Fixing return statement for arrow functions, make it more simple

Created by: Maxteh

Подправил стрелочные функции, так гораздо удобнее воспринимать информацию, чем с лишними return