Update create function to allow other elements