Add ability to pass in width + height