code for a few easy problems