Final code for submission with kwargs