Simplify code. #mania