gotta work on logic