I did it a bit to complicated, reduced lines of code