the code is beautiful now